# @Author：cnzdy
# @Email：cn_zdy@126.com
# @Time: 2021/10/9 10:47
# @File: options.py

from pytorch.classifier.models import AlexNet


class Options(object):
    """
        用类保存程序的运行参数信息。
    """
    env = 'default'  # visdom 环境
    vis_port = 8097  # visdom 端口
    project_name = 'cat dog classification'
    model_name = 'AlexNet'
    model = AlexNet()
    class_names = ['Cat', 'Dog']

    root = r'M:\classification\cat-dog\data\mini'
    models_path = r'M:/classification/cat-dog/models/'
    model_file = r"M:\classification\cat-dog\models\cat dog classification_AlexNet_best.pth"
    test_result_file = r'M:\classification\cat-dog\analysis\result.csv'
    analysis_path = r'M:\classification\cat-dog\analysis'

    train_loader = None
    val_loader = None
    test_loader = None

    num_classes = 2
    batch_size = 8
    num_workers = 8

    epochs = 10
    lr = 0.0001
    # 学习率衰减，when val_loss increase, lr=lr*0.95
    lr_decay = 0.95